#include "myhead.h"
#include <iostream>
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/wait.h>
#include <sys/wait.h>
#include <signal.h>
#include <sys/ipc.h>
#include <sys/msg.h>
#include <sys/shm.h>
#include <sys/sem.h>
#include <pthread.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <netinet/in.h>
using namespace std;
#define SER_PORT 8080           // 服务器端口号
#define SER_IP "192.168.85.131" // 服务器IP地址
/**********************主程序************************/
void handler(int signo)
{
    if (signo == SIGCHLD)
    {
        while (waitpid(-1, NULL, WNOHANG) > 0)
            ; // 循坏以非阻塞形式回收僵尸进程
    }
}
int main(int argc, char const *argv[])
{
    // 将子进程发送

    // 1.创建用于连接的套接字的描述符
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    // 参数1.AF_INET表示使用的是IPv4的通信协议
    // 参数2.SOCK_STREAM使用的是TCP通信
    // 参数3.由于参数2指定了协议,参数3填0即可
    if (sfd == -1)
    {
        perror("socket error");
        return -1;
    }
    printf("socket success sfd=%d\n", sfd); // 3
    // 2.绑定ip地址和端口号
    // 2.1填充要绑定的ip地址和端口号结构体
    struct sockaddr_in sin;
    sin.sin_family = AF_INET;         // 通信域
    sin.sin_port = htons(SER_PORT);   // 端口号
    sin.sin_addr.s_addr = INADDR_ANY; // ip地址
                                      // 2.2绑定工作
                                      // 参数1:要被绑定的套接字文件描述符
                                      // 参数2:要绑定的信息结构体,需要进行强制类型转换,防止警告
                                      // 参数3:参数2的大小
    if (bind(sfd, (struct sockaddr *)&sin, sizeof(sin)) == -1)
    {
        perror("bind error");
        return -1;
    }
    // 3.启动监听
    // 参数1:要启动监听的文件描述符
    // 参数2:要挂起队列的长度
    if (listen(sfd, 128) == -1)
    {
        perror("listen error");
        return -1;
    }
    printf("listen success\n");
    // 4.阻塞等待客户端的连接请求
    struct sockaddr_in cin;
    socklen_t socklen = sizeof(cin);
    while (1)
    {
        int newfd = accept(sfd, (struct sockaddr *)&cin, NULL);
        // 参数1:服务器接套接字文件描述符
        // 参数2:用于接收客户端地址信息结构体的容器,如果不接收,也可以NULL
        // 参数3:用于接收参数2的大小,如果参数2为NULL,则参数3也是NULL
        if (newfd == -1)
        {
            perror("accept error");
            return -1;
        }
        printf("[%s:%d]:有新的客户端连接成功,newfd=%d!!!!\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);
        // 创建子进程,用于跟客户端进行通信
        pid_t pid = fork();
        if (pid > 0)
        {
            // 父进程内容
            // wait(NULL); // 不可以使用阻塞函数
            // waitpid(-1, NULL, WNOHANG); // 以非阻塞形式回收?
            // 关闭newfd因为,父进程不为newfd,
        }
        else if (pid == 0)
        {
            // 对收到的数据处理一下
            // 5.数据收发
            char rbuf[128] = ""; // 数据容器
            while (1)
            {
                // 关闭sfd,因为子进程进行客户端通信,主要还是使用newfd
                //  清空容器中的内容
                bzero(rbuf, sizeof(rbuf));
                // 从套接字中读取消息
                int res = recv(newfd, rbuf, sizeof(rbuf), 0);
                if (res == -1)
                {
                    perror("recv error");
                    break;
                }
                else if (res == 0)
                {
                    printf("对端已经下线");
                    break;
                }
                printf("收到数据:%s\n", rbuf);
                strcat(rbuf, "* *");
                if (send(newfd, rbuf, strlen(rbuf), 0) == -1)
                {
                    perror("send error");
                    return -1;
                }
                printf("发送成功");
            }
            // 6.关闭套接字
            close(newfd);
        }
        else
        {
            perror("fork error");
            return -1;
        }
    }
    close(sfd);
    return 0;
}
